package com.hc.programming.linked;

import com.hc.programming.bean.ListNode;

import java.util.Arrays;

/**
 * 给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回 新的头节点 。
 * <p>
 * 示例 1：
 * 输入：head = [1,2,6,3,4,5,6], val = 6
 * 输出：[1,2,3,4,5]
 * 示例 2：
 * 输入：head = [], val = 1
 * 输出：[]
 * 示例 3：
 * 输入：head = [7,7,7,7], val = 7
 * 输出：[]
 * <p>
 * 提示：
 * 列表中的节点数目在范围 [0, 10^4] 内
 * 1 <= Node.val <= 50
 * 0 <= val <= 50
 *
 * @author huangchao E-mail:fengquan8866@163.com
 * @version 创建时间：2024/10/23 19:37
 */
public class 移除链表元素 {
    public static void main(String[] args) {
        System.out.println(removeElements(new ListNode(Arrays.asList(1, 2, 6, 3, 4, 5, 6)), 6));
        System.out.println(removeElements(new ListNode(Arrays.asList()), 1));
        System.out.println(removeElements(new ListNode(Arrays.asList(7, 7, 7, 7)), 7));
    }

    public static ListNode removeElements(ListNode head, int val) {
//        return 一次遍历(head, val);
        return 虚拟头结点(head, val);
    }

    private static ListNode 虚拟头结点(ListNode head, int val) {
        if (head == null) return null;
        ListNode dummy = new ListNode();
        dummy.next = head;
        ListNode pre = dummy;
        while (pre.next != null) {
            if (pre.next.val == val) {
                pre.next = pre.next.next;
            } else {
                pre = pre.next;
            }
        }
        return dummy.next;
    }

    private static ListNode 一次遍历(ListNode head, int val) {
        if (head == null) return null;
        ListNode pre = null;
        ListNode cur = head;
        while (cur!= null) {
            if (cur.val == val) {
                if (pre == null) {
                    head = cur.next;
                } else {
                    pre.next = cur.next;
                }
            } else {
                pre = cur;
            }
            cur = cur.next;
        }
        return head;
    }
}
